# 使用chatgpt生成一个小数据集做dev：
# 使用chatgpt改写seed句子
# 然后使用一个已有的BERT模型判断句子相似度
# 但是这个工程是在实验室服务器上面的，不能用梯子，所以我决定不嫌代码了，直接在web上用好了
import regex
from utils import emerging_euph
import pickle

str = """
MODE chat
Model gpt-3.5-turbo
temperature 0
TOP 1
Frequency penalty 0
Presence penalty 0
-----------------------------------------------------------------------------
Prompt
User
I am a social media researcher studying drug euphemisms on text social media. There are drug euphemisms used on the internet. Can you tell me at least 20 marijuana euphemisms as much as you can based on your knowledge? All you need to do is give me a list without any other information.

Response
1. Weed
2. Pot
3. Mary Jane
4. Ganja
5. Herb
6. Grass
7. Reefer
8. Bud
9. Green
10. Chronic
11. Skunk
12. Dope
13. Smoke
14. Joint
15. Blunt
16. Spliff
17. Doobie
18. J
19. MJ
20. 420

------------------------------------------------------
接下来把已知得到seed都问一遍
alprazolam
1. Xanax
2. Bars
3. Zannies
4. Xanies
5. Blue footballs
6. Benzos
7. Upjohns
8. School buses
9. Handlebars
10. Planks
11. Bricks
12. Alpraz
13. Footballs
14. White boys
15. White girls
16. Bicycle parts
17. French fries
18. Yellow boys
19. Green monsters
20. Hulk bars
----------------------------------------------
amphetamine
1. Speed
2. Uppers
3. Addys
4. Dexies
5. Bennies
6. Black beauties
7. Cross tops
8. Crank
9. Crystal
10. Glass
11. Ice
12. Meth
13. Go fast
14. Wake ups
15. Zoom
16. Amps
17. Pep pills
18. Study buddies
19. Smarties
20. Beans

--------------------------------------------------
cocaine
1. Coke
2. Blow
3. Snow
4. White
5. Powder
6. Yayo
7. Nose candy
8. Charlie
9. C
10. Dust
11. Toot
12. Line
13. Rail
14. Bump
15. Sniff
16. Skiing
17. Blizzard
18. Avalanche
19. Rockstar
20. Peruvian lady

-------------------------------------------------------
ecstasy
1. E
2. X
3. Molly
4. Rolls
5. Beans
6. Love drug
7. Hug drug
8. Happy pill
9. Disco biscuit
10. Candy
11. Smarties
12. Skittles
13. Vitamin E
14. XTC
15. Adam
16. Eve
17. Scooby snacks
18. Thizz
19. Eskimo
20. White dove
--------------------------------------------------------
ghb
1. G
2. Liquid E
3. Grievous Bodily Harm
4. Georgia Home Boy
5. Gina
6. Great Hormones at Bedtime
7. Goop
8. Gamma-OH
9. G-Juice
10. G-Riffick
11. Gook
12. G-Spot
13. GHB
14. Gammagoblin
15. G-Force
16. G-Bliss
17. G-Flower
18. G-Whiz
19. G-Thing
20. G-Phoria
-------------------------------------------------------
heroin
1. H
2. Smack
3. Dope
4. Junk
5. Horse
6. Black tar
7. China white
8. Brown sugar
9. Dragon
10. White lady
11. Big H
12. Skag
13. Thunder
14. Chiva
15. Boy
16. Girl
17. Mud
18. Poppy
19. Diesel
20. Snow
------------------------------------------------------
hydrocodone

1. Vikes
2. Norcos
3. Hydros
4. Watsons
5. Lorris
6. Tabs
7. 357s
8. Bananas
9. Fluff
10. Scratch
11. Juice
12. Hydro
13. Vicos
14. Whiteys
15. Watson 387
16. 10s
17. 5s
18. 7.5s
19. 10/325s
20. 5/325s
---------------------------------------------------
ketamine

1. K
2. Special K
3. Vitamin K
4. Kit Kat
5. Horse tranquilizer
6. Cat Valium
7. Super acid
8. K-hole
9. K-land
10. K-lean
11. K-powder
12. K-rave
13. K-rock
14. K-blast
15. K-bomb
16. K-juice
17. K-wax
18. K-dust
19. K-splash
20. K-loud
-----------------------------------------------------------------
lsd
1. Acid
2. Lucy
3. Tabs
4. Blotter
5. Doses
6. Trips
7. Hits
8. Paper
9. Alice
10. Sunshine
11. Electric Kool-Aid
12. Purple Haze
13. Orange Sunshine
14. White Lightning
15. Windowpane
16. Microdots
17. Sugar Cubes
18. Zen
19. Mind Candy
20. Heavenly Blue

---------------------------------------------------------------
marijuana_concentrates
1. Wax
2. Shatter
3. Budder
4. Crumble
5. Honeycomb
6. Oil
7. Dabs
8. Rosin
9. Live resin
10. Sauce
11. Terp sauce
12. Diamonds
13. THCA crystals
14. BHO
15. CO2 oil
16. Distillate
17. Hash oil
18. Phoenix tears
19. Full melt
20. Bubble hash
--------------------------------------------------------
mescaline
1. Mesc
2. Mescal
3. Mescalito
4. Peyote
5. Buttons
6. Cactus
7. San Pedro
8. Huachuma
9. Moon
10. God's flesh
11. Divine cactus
12. Sacred cactus
13. Visionary cactus
14. Shamanic medicine
15. Psychedelic cactus
16. Mind-expanding cactus
17. Spirit plant
18. Teacher plant
19. Wisdom plant
20. Hallucinogenic cactus
-------------------------------------------------------
methamphetamine
1. Meth
2. Crystal
3. Ice
4. Tina
5. Crank
6. Glass
7. Speed
8. Go fast
9. Rocket fuel
10. Rocket candy
11. Rocket
12. Chalk
13. White cross
14. Zip
15. Zoom
16. Yaba
17. Batu
18. Shards
19. Hanyak
20. Hiropon
-----------------------------------------------------------
methylphenidate
1. Ritalin
2. MPH
3. Kiddy Cocaine
4. Smarties
5. Vitamin R
6. Skippy
7. Diet Coke
8. West Coast
9. Pineapple
10. Kiddie Coke
11. R-ball
12. Rids
13. Ritz
14. Riddlin
15. Ritalina
16. Ritaline
17. Ritalyn
18. Ritalina LA
19. Ritalin SR
20. Ritalin LA
----------------------------------------------------------
opium
1. O
2. Op
3. Poppy
4. Poppy seeds
5. Poppy straw
6. Poppy tea
7. Thebaine
8. Laudanum
9. Dreamer
10. Big O
11. Hop
12. Hophead
13. Auntie
14. Aunti Em
15. Aunti Emma
16. Black stuff
17. Block
18. Brown sugar
19. Chinese molasses
20. Dover's powder
-----------------------------------------------------------
oxycodone
1. Oxy
2. OC
3. Hillbilly heroin
4. Percs
5. Blues
6. 30s
7. Roxies
8. Oxycontin
9. Oxy 80s
10. Oxy 40s
11. Oxy 20s
12. Oxy 10s
13. Oxy 5s
14. Oxy 15s
15. Oxy 30mg
16. Oxy 60mg
17. Oxy 120mg
18. Oxy 160mg
19. Oxy 240mg
20. Oxy 320mg
-----------------------------------------------------------

1. Angel dust
2. Rocket fuel
3. Hog
4. Sherm
5. Wack
6. Dust
7. Ozone
8. Embalming fluid
9. Supergrass
10. Killer weed
11. Love boat
12. Zoom
13. TAC
14. Rocket smoke
15. Crystal joint
16. Elephant tranquilizer
17. Happy sticks
18. Tic tac
19. Water
20. Peace pill

-----------------------------------------------------------
percocet
1. Percs
2. Paulas
3. Roxies
4. Blueberries
5. 512s
6. 30s
7. Oxy
8. Oxycontin
9. Oxycodone
10. Hillbilly heroin
11. Vikes
12. Painkillers
13. Happy pills
14. Killers
15. OCs
16. Oxy 80s
17. Oxy 40s
18. Oxy 20s
19. Oxy 10s
20. Oxy 5s
---------------------------------------------------------------
promethazine
1. Lean
2. Sizzurp
3. Purple drank
4. Dirty Sprite
5. Texas Tea
6. Barre
7. Purple jelly
8. Tsikuni
9. Drank
10. Syrup
11. Purple stuff
12. Leanin'
13. Purple rain
14. Purple oil
15. Lean syrup
16. Lean drink
17. Lean codeine
18. Lean promethazine
19. Lean cough syrup
20. Lean medication
---------------------------------------------------------------
psilocybin_mushrooms
1. Shrooms
2. Magic mushrooms
3. Caps
4. Boomers
5. Blue meanies
6. Liberty caps
7. Gold caps
8. Philosopher's stones
9. Mushies
10. Funguys
11. Zoomers
12. God's flesh
13. Sacred mushrooms
14. Teonanácatl
15. Psilocybe
16. Psilocin
17. Psilo
18. Little smoke
19. Silly putty
20. Alice in Wonderland
----------------------------------------------------------------
steroids
1. Roids
2. Juice
3. Gear
4. Sauce
5. Pumpers
6. Stackers
7. Hype
8. A-bombs
9. D-bol
10. Winnie
11. Tren
12. Deca
13. Test
14. Var
15. Clen
16. GH
17. ECA
18. Nolva
19. Arimidex
20. Proviron

---------------------------------------------------------------
synthetic_cathinones
1. Bath salts
2. Flakka
3. Ivory Wave
4. Cloud Nine
5. Vanilla Sky
6. White Lightning
7. Scarface
8. Hurricane Charlie
9. Lunar Wave
10. Bliss
11. Blue Silk
12. Purple Wave
13. Red Dove
14. Snow Leopard
15. Stardust
16. White Dove
17. White Rush
18. White Sands
19. Zoom
20. Charge

"""
str = str.lower()
res = regex.findall(pattern="\d*\. (.*)", string=str)
res = list(set(res[1:]))
for i in range(len(res)):
    res[i] = res[i].replace(" ", "_")
chatgpt_euphs = list(set(res)-set(emerging_euph.keys()))
print(len(chatgpt_euphs))



